# coding: utf-8
import os
from datetime import datetime


def get_folder_size(folder):
    """
    获取文件夹大小（以字节为单位）。
    :param folder: 文件夹路径。
    :return: 文件夹大小（字节）。
    """
    total_size = 0
    for dir_path, dir_names, filenames in os.walk(folder):
        for f in filenames:
            fp = os.path.join(dir_path, f)
            try:
                total_size += os.path.getsize(fp)
            except OSError:
                # 可能是某些文件无法访问，如链接，设备文件等
                pass
    kb = total_size / 1024
    mb = kb / 1024
    gb = mb / 1024
    if kb < 1024:
        return f'{round(kb, 3)} KB'
    elif mb < 1024:
        return f'{round(mb, 3)} MB'
    else:
        return f'{round(gb, 3)} GB'


def getTimestamp():
    """
    获取当前时间戳（秒）。
    :return: 时间戳（秒）。
    """
    return datetime.now().strftime('%Y%m%d%H%M%S')
